package com.atguigu.rabbitmq.four;


import com.atguigu.rabbitmq.util.RabbitMqUtils;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.ConfirmCallback;

import java.util.UUID;
import java.util.concurrent.ConcurrentNavigableMap;
import java.util.concurrent.ConcurrentSkipListMap;

/**
 * 验证发布确认模式
 *  使用的时间  比较那种确认方式时最好的
 *  1：单个确认模式
 *  2：批量确认模式
 *  3：异步批量确认
 */
public class ConfirmMessage {

    // 批量发消息的个数
    public static final int MESSAGE_COUNT = 1000;

    public static void main(String[] args) throws Exception {
        // 单个确认 耗时1056ms
        // publishMessageIndividually();
        // 批量确认 耗时144ms
        // publishMessageBatch();
        // 异步确认 耗时122ms,打印的比较多 性能最高
        publishMessageAsync();
    }

    /**
     * 单个确认
     */
    public static void publishMessageIndividually() throws Exception {
        Channel channel = RabbitMqUtils.getChannel();
        // 队列的声明
        String queueName = UUID.randomUUID().toString();

        // 声明队列
        channel.queueDeclare(queueName,false,false,false,null);
        // 确认发布确认
        channel.confirmSelect();
        // 记录开始的时间
        long begin = System.currentTimeMillis();

        // 批量发消息
        for ( int i = 0 ; i < MESSAGE_COUNT ; i ++ ) {
            String message = i+"";
            channel.basicPublish("",queueName,null,message.getBytes());
            // 单个消息等候确认
            boolean flag = channel.waitForConfirms();
            if ( flag ) {
                System.out.println("消息发送成功");
            }
        }

        // 结束时间
        long end = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_COUNT+"个单独确认消息，耗时"+( end - begin ) + "ms");
    }

    /**
     * 批量发布确认
     */
    public static void publishMessageBatch() throws Exception {
        Channel channel = RabbitMqUtils.getChannel();
        // 队列的声明
        String queueName = UUID.randomUUID().toString();

        // 声明队列
        channel.queueDeclare(queueName,false,false,false,null);
        // 确认发布确认
        channel.confirmSelect();
        // 记录开始的时间
        long begin = System.currentTimeMillis();

        // 批量确认消息的大小
        int batchSize = 100;
        // 未确认消息的个数

        // 批量发送消息 批量发布确认
        for ( int i = 0 ; i < MESSAGE_COUNT ; i ++ ) {
            String message = i+"";
            channel.basicPublish("",queueName,null,message.getBytes());

            // 判断消息发送到100条的时候，批量确认一次
            if ( i % batchSize == 0 ) {
                channel.waitForConfirms();
            }
        }

        // 结束时间
        long end = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_COUNT+"个批量确认消息，耗时"+( end - begin ) + "ms");
    }

    /**
     * 异步确认发送耗时多久
     */
    public static void publishMessageAsync() throws Exception {
        Channel channel = RabbitMqUtils.getChannel();
        // 队列的声明
        String queueName = UUID.randomUUID().toString();

        // 声明队列
        channel.queueDeclare(queueName,false,false,false,null);
        // 确认发布确认
        channel.confirmSelect();
        // 记录开始的时间
        long begin = System.currentTimeMillis();


        /**
         *  线程安全有序的一个哈希表 适用于高并发的情况下
         *  1.轻松的将序号与消息进行关联
         *  2.轻松删除条目 只要给到序号
         *  3. 支持高并发（高并发）
         */
        ConcurrentSkipListMap<Long,String> outstandingConfirms = new ConcurrentSkipListMap<>();

        // 消息确认成功 回调函数
        ConfirmCallback ascCallback = ( deliveryTag,multiple) -> {
            // 2：删除掉已经确认的消息 剩下的就是未确认的消息
            if ( multiple ) { // 是否批量
                ConcurrentNavigableMap<Long, String> confirmed = outstandingConfirms.headMap(deliveryTag);
                confirmed.clear(); // 删除
            } else {
                outstandingConfirms.remove(deliveryTag);
            }


            System.out.println("确认消息的标号："+deliveryTag);
        };


        // 消息确认失败 回调函数
        /**
         * 1：消息的标记
         * 2：是否未批量确认
         */
        ConfirmCallback nackCallback = ( deliveryTag,multiple) -> {
            // 打印未确认的消息都有那些
            String message = outstandingConfirms.get(deliveryTag);
            System.out.println("未确认的消息是："+message +"，未确认的小时tag:"+deliveryTag);
        };
        // 准备消息的监听器，监听那些消息成功了，那些消息失败了
        /**
         * 1:监听那些消息成功了
         * 2：监听那些消息失败了
         */
        channel.addConfirmListener(ascCallback,nackCallback);
        for ( int i = 0 ; i < MESSAGE_COUNT ; i ++ ) {
            String message = "消息："+i;
            channel.basicPublish("",queueName,null,message.getBytes());

            // 此处记录下所有要发送的消息  消息的总和
            outstandingConfirms.put(channel.getNextPublishSeqNo(),message);
        }

        // 结束时间
        long end = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_COUNT+"个批量确认消息，耗时"+( end - begin ) + "ms");

    }

}
